<template>
    <div class="add-role">
        <header class="add-header">
            <div class="left">
                <span class="arrow" @click="$router.back()"><i class="el-icon-arrow-left" />返回</span>
                <span>|</span>
                <span>{{ roleId ? '编辑角色' : '添加角色' }}</span>
            </div>
            <div class="right">
                黑马程序员
            </div>
        </header>
        <main class="add-main">
            <div class="step-container">
                <el-steps direction="vertical" :active="activeStep">
                    <el-step title="角色信息" />
                    <el-step title="权限信息" />
                    <el-step title="检查并完成" />
                </el-steps>
            </div>
            <div class="form-container" v-show="activeStep === 1">
                <div class="title">角色信息</div>
                <div class="form">
                    <el-form class="form-box" ref="roleForm" :model="roleForm" :rules="roleRules">
                        <el-form-item label="角色名称" prop="roleName">
                            <el-input v-model="roleForm.roleName" />
                        </el-form-item>
                        <el-form-item label="角色描述">
                            <el-input v-model="roleForm.remark" />
                        </el-form-item>
                    </el-form>
                </div>
            </div>
            <div class="form-container" v-show="activeStep === 2">
                <div class="title">权限配置</div>
                <div class="form">
                    <div class="tree-wrapper">
                        <div v-for="item in treeList" :key="item.id" class="tree-item">
                            <div class="tree-title">{{ item.title }}</div>
                            <el-tree ref="tree" :data="item.children" show-checkbox default-expand-all node-key="id"
                                highlight-current :props="{ label: 'title' }" check-strictly />
                        </div>
                    </div>
                </div>
            </div>
            <div class="form-container" v-show="activeStep === 3">
                <div class="title">检查并完成</div>
                <div class="form">
                    <div class="info">
                        <div class="from-item">角色名称：{{ roleForm.roleName }}</div>
                        <div class="from-item">角色描述：{{ roleForm.remark }}</div>
                        <div class="from-item">角色权限：</div>
                        <div class="tree-wrapper">
                            <div v-for="item in disableTreeList" :key="item.id" class="tree-item">
                                <div class="tree-title">{{ item.title }}</div>
                                <el-tree ref="diabledTree" :data="item.children" show-checkbox default-expand-all
                                    node-key="id" :highlight-current="false" :props="{ label: 'title' }"
                                    check-strictly />
                            </div>
                        </div>
                    </div>
                </div>
            </div>
        </main>
        <footer class="add-footer">
            <div class="btn-container">
                <el-button v-if="activeStep > 1" @click="decreseStep">上一步</el-button>
                <el-button v-if="activeStep < 3" type="primary" @click="increseStep">下一步</el-button>
                <el-button v-if="activeStep === 3" type="primary" @click="confirmAdd">
                    {{ roleId ? '确认修改' : '确认添加' }}
                </el-button>
            </div>
        </footer>
    </div>
</template>
<script>
import { getTreeListAPI, createRoleUserAPI, getRoleDetailAPI, updateRoleAPI } from '@/api/role'
export function addDisabled(treeList) {
    treeList.forEach(item => {
        item.disabled = true
        if (item.children && item.children.length > 0) {
            addDisabled(item.children)
        }
    })
}
export default {
    data() {
        return {
            roleId: this.$route.params.id,
            activeStep: 1,
            roleForm: {
                roleName: '',
                remark: '',
                perms: []
            },
            roleRules: {
                roleName: [
                    { required: true, message: '请输入角色名称', trigger: 'blur' }
                ]
            },
            treeList: [],
            disableTreeList: []
        }
    },
    methods: {
        decreseStep() {
            if (this.activeStep === 1) return
            this.activeStep--
        },
        increseStep() {
            if (this.activeStep === 3) return
            if (this.activeStep === 1) {
                this.$refs.roleForm.validate(valid => {
                    if (valid) {
                        this.activeStep++
                    }
                })
            } else if (this.activeStep === 2) {
                this.roleForm.perms = []
                this.$refs.tree.forEach(item => {
                    this.roleForm.perms.push(item.getCheckedKeys())
                });
                if (this.roleForm.perms.flat().length === 0) {
                    this.$message.error('请至少选择一个权限点')
                } else {
                    this.activeStep++
                    //回填已选择数据
                    this.$refs.diabledTree.forEach((item, index) => {
                        item.setCheckedKeys(this.roleForm.perms[index])
                    })
                }
            }
        },
        async getTreeList() {
            const res = await getTreeListAPI()
            this.treeList = res.data
            //深拷贝
            this.disableTreeList = JSON.parse(JSON.stringify(this.treeList))
            addDisabled(this.disableTreeList)
        },
        async getRoleDetail() {
            try {
                const res = await getRoleDetailAPI(this.roleId)
                // 回填表单数据
                this.roleForm.roleName = res.data.roleName
                this.roleForm.remark = res.data.remark
                
                // 等待权限树加载完成后回填权限数据
                await this.$nextTick()
                if (this.$refs.tree) {
                    this.$refs.tree.forEach((item, index) => {
                        item.setCheckedKeys(res.data.perms[index])
                    })
                }
            } catch (error) {
                console.error('获取角色详情失败:', error)
                this.$message.error('获取角色详情失败')
            }
        },
        async confirmAdd() {
            try {
                if (this.roleId) {
                    // 编辑角色
                    await updateRoleAPI({
                        ...this.roleForm,
                        roleId: this.roleId
                    })
                    this.$message({
                        type: 'success',
                        message: '编辑角色成功'
                    })
                } else {
                    // 新增角色
                    await createRoleUserAPI(this.roleForm)
                    this.$message({
                        type: 'success',
                        message: '添加角色成功'
                    })
                }
                this.$router.back()
            } catch (error) {
                console.error('操作失败:', error)
                this.$message.error('操作失败')
            }
        }
    },
    async mounted() {
        // 先获取权限树，再获取角色详情
        await this.getTreeList()
        if (this.roleId) {
            await this.getRoleDetail()
        }
    }
}
</script>
<style scoped lang="scss">
.add-role {
    background-color: #f4f6f8;
    height: 100vh;

    .add-header {
        display: flex;
        justify-content: space-between;
        padding: 0 20px;
        height: 64px;
        line-height: 64px;
        font-size: 16px;
        font-weight: bold;
        background-color: #fff;

        .left {
            span {
                margin-right: 4px;
            }

            .arrow {
                cursor: pointer;
            }
        }

        .right {
            text-align: right;
        }
    }

    .add-main {
        position: fixed;
        top: 64px;
        bottom: 88px;
        width: 100%;
        overflow-y: auto;
        background: #f4f6f8;
        padding: 20px 230px;
        display: flex;

        .step-container {
            height: 400px;
            width: 200px;
        }

        .form-container {
            flex: 1;
            background-color: #fff;
            overflow-y: auto;

            .title {
                height: 60px;
                line-height: 60px;
                padding-left: 20px;
            }

            .form {
                margin-bottom: 20px;
                padding: 20px 65px 24px;

                .el-form {
                    display: flex;
                    flex-wrap: wrap;
                }

                .info {
                    font-size: 14px;
                    color: #666;

                    .form-item {
                        margin-bottom: 20px;
                    }
                }
            }

            .form-box {
                width: 600px;
                display: flex;
                flex-direction: column;
            }
        }
    }

    .add-footer {
        position: fixed;
        bottom: 0;
        width: 100%;
        padding: 24px 50px;
        color: #000000d9;
        font-size: 14px;
        background: #fff;
        text-align: center;
        z-index: 10001;
    }

    .tree-wrapper {
        display: flex;
        justify-content: space-between;
        flex-wrap: wrap;

        .tree-item {
            flex: 1;
            padding: 0px 4px;
            text-align: center;

            .tree-title {
                background: #f5f7fa;
                text-align: center;
                padding: 10px 0;
                margin-bottom: 12px;
            }
        }
    }
}
</style>